from data_load import DataManager
from FLServer import FLServer

# 为了保证数据规模与集中训练一直，保证client_number*data_counts=60000 
client_number = 20
data_counts = 3000

# 初始化，并加载数据
dm = DataManager(client_number=20)
dm.load_data()

# # 初始化，并开始训练
# server = FLServer(dm.allocate_data_avg(),dm.allocate_data_test())
# server.start_federated()

# # 初始化并开始训练
# server = FLServer(dm.allocate_data_iid(data_counts=3000),dm.allocate_data_test())
# server.start_federated()

# 初始化并开始训练
server = FLServer(dm.allocate_data_noniid(class_counts=2,data_counts=3000),dm.allocate_data_test())
server.start_federated() 
